Reduce data transfer in Spark